"""
lstm模型配置
"""


class LSTMConfig(object):
    def __init__(self):
        self.epochs = 100    # 训练代数
        self.save_epoch=10   # 每隔多少代保存一次
        self.batch_size = 64     # 批处理大小
        self.max_seq_len = 29    # 序列最大长度
        self.feature_len=48      # 特征长度
        self.num_layers=1         # 隐藏层数量
        self.hidden_size = 128   # 隐藏层维度
        self.outputs_size = 3    # 输出层维度
        self.learning_rate = 1e-4  # 学习率
        self.input_vec_len = 48  # 输入向量的维度
        self.dropout=0.9         # dropout
        self.clip_grad=None         # 梯度修剪
        self.bidirectional=True  # 网络结构,True: net为双向LSTM False: net为单向LSTM
        self.data_dir=r"/home/lisen/tool/PyProjects/唇语识别/dataset/dataset_word/dataset_v2"  # 数据集目录
        self.weight_save_dir=r"/home/lisen/tool/PyProjects/唇语识别/weight_files/LSTM_weight/weight_file_v2_128_1_09"
        self.continue_train=False   # 是否进行增量学习
        self.tmp_dir=r"/home/lisen/tool/PyProjects/唇语识别/tmp"


lstm_config=LSTMConfig()
